package com.chen.websocket;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.CopyOnWriteArraySet;

import javax.servlet.http.HttpSession;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;

import com.alibaba.fastjson.JSON;

import net.sf.json.JSONObject;
 
//该注解用来指定一个URI，客户端可以通过这个URI来连接到WebSocket。类似Servlet的注解mapping。无需在web.xml中配置。
@ServerEndpoint("/chat")
public class MyWebSocket {
    //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
    private static int onlineCount = 0;
    //concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
    private static CopyOnWriteArraySet<MyWebSocket> webSocketSet = new CopyOnWriteArraySet<MyWebSocket>();
    //与某个客户端的连接会话，需要通过它来给客户端发送数据
    private Session session;
    private static Map<String,String> sessionsMap = new HashMap<>();
    
    /**
     * 连接建立成功调用的方法
     * @param session  可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
     */
    @OnOpen
    public void onOpen(Session session){
    	
        this.session = session;
        Map<String, List<String>> map = session.getRequestParameterMap();
        
        boolean islogin = true;
        for (Entry<String, String> entry : getSessionsMap().entrySet()) {  
            System.out.println("Key = " + entry.getKey() + ", Value = " + entry.getValue());  
            if(entry.getValue().equals(map.get("userid").get(0))){
            	islogin = false;
            	break;
            }
        }
        if(islogin){
        	putSessionsMap(session.getId(), map.get("userid").get(0));
            webSocketSet.add(this);     //加入set中
            addOnlineCount();           //在线数加1
        }
        System.out.println("连接用户：");
        
        System.out.println("当前连接用户："+map.get("userid").get(0));
        System.out.println("有新连接加入！当前在线人数为" + getOnlineCount());
    }
     
    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose(){
        webSocketSet.remove(this);  //从set中删除
        System.out.println(this.session.getId());
        removeSessionsMap(this.session.getId());
        subOnlineCount();           //在线数减1    
        System.out.println("有一连接关闭！当前在线人数为" + getOnlineCount());
    }
     
    /**
     * 收到客户端消息后调用的方法
     * @param message 客户端发送过来的消息
     * @param session 可选的参数
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        System.out.println("来自客户端的消息:" + message);
        
        
        //String jsonstring = "{content:51,toUser:18150166685}";      
        Message msg = JSON.parseObject(message,Message.class);  

        System.out.println(msg);

        //群发消息
        for(MyWebSocket item: webSocketSet){ 
        	String userid = getSessionsMap().get(item.session.getId());
        	//检测该用户是否为要发送的用户
        	if(userid.equals(msg.getToUser())){
        		try {
                    item.sendMessage("发送给"+msg.getToUser()+"的信息:"+msg.getContent());
                    break;
                } catch (IOException e) {
                    e.printStackTrace();
                    continue;
                }
        	}
        	continue;
        }
    }
     
    /**
     * 发生错误时调用
     * @param session
     * @param error
     */
    @OnError
    public void onError(Session session, Throwable error){
        System.out.println("发生错误");
        error.printStackTrace();
    }
     
    /**
     * 这个方法与上面几个方法不一样。没有用注解，是根据自己需要添加的方法。
     * @param message
     * @throws IOException
     */
    public void sendMessage(String message) throws IOException{
        this.session.getBasicRemote().sendText(message);
        //this.session.getAsyncRemote().sendText(message);
    }
 
    public static synchronized int getOnlineCount() {
        return onlineCount;
    }
 
    public static synchronized void addOnlineCount() {
        MyWebSocket.onlineCount++;
    }
     
    public static synchronized void subOnlineCount() {
        MyWebSocket.onlineCount--;
    }

	public static synchronized Map<String,String> getSessionsMap() {
		return MyWebSocket.sessionsMap;
	}

	public static synchronized void putSessionsMap(String key,String value) {
		MyWebSocket.sessionsMap.put(key, value);
	}
	
	public static synchronized void removeSessionsMap(String key) {
		MyWebSocket.sessionsMap.remove(key);
	}


}

